package com.ruoyi.base.redis.login;

import com.ruoyi.base.utils.ip.IpUtils;
import com.ruoyi.base.utils.servlet.ServletUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.util.concurrent.TimeUnit;

/**
 * 密码输入错误次数
 *
 * @author lcf
 */
@Component
public class PwdWrongTimesRedis {
    // 令牌有效期（默认30分钟）
    @Value("${token.expireTime}")
    private int expireTime;
    @Autowired
    public StringRedisTemplate redisTemplate;

    /**
     * key前缀
     */
    private final String prefix = "pwdWrongTimes:";
    /**
     * 过期时间单位
     */
    private TimeUnit timeUnit = TimeUnit.HOURS;

    /**
     * 删除缓存
     */
    public void delCache() {
        redisTemplate.delete(getKey());
    }

    /**
     * 缓存
     */
    public void cache() {
        int times = 1;
        String cache = getCache();
        if (StringUtils.isNotBlank(cache)) {
            times += Integer.parseInt(cache);
        }

        redisTemplate.boundValueOps(getKey()).set(String.valueOf(times), expireTime - 1, timeUnit);
    }

    /**
     * 获取
     */
    public String getCache() {
        return redisTemplate.boundValueOps(getKey()).get();
    }

    /**
     * 检查密码错误是否大于等于3次
     */
    public boolean isNeedVerifyCode() {
        String cache = getCache();
        if (StringUtils.isBlank(cache)) {
            return false;
        }

        return Integer.parseInt(cache) >= 3;
    }

    /**
     * 拼接key
     */
    private String getKey() {
        return prefix + IpUtils.getIpAddr(ServletUtils.getRequest());
    }
}
